#常用工具

from functools import wraps
import time

DEBUG = True


def calculate_time(func):
    @wraps(func)  # 保留原始函数的元信息
    def wrapper(*args, **kwargs):
        if DEBUG:
            start_time = time.time()  # 记录开始时间
            result = func(*args, **kwargs)  # 执行函数
            end_time = time.time()  # 记录结束时间
            GREEN = "\033[32m"
            RESET = "\033[0m"  # 重置颜色
            print(f"{GREEN}以上函数 {func.__name__} 运行时间: {end_time - start_time:.4f} 秒{RESET}")
            return result
        else:
            result = func(*args, **kwargs)  # 执行函数
            return result
    return wrapper


if __name__ == '__main__':
    print(1)

